#include "UdpServer.hpp"
#include <memory>
#include <vector>

void Usage(std::string proc)
{
    std::cout << "\n\rUsage: " << proc << " port[1024+]\n" << std::endl;
}
std::string Handler(const std::string &str)
{
    std::string res = "Server get a message: ";
    res += str;
    std::cout << res << std::endl;

    return res;
}
// 判断指令是否安全
bool SafeCheck(const std::string &cmd)
{
    int safe = false;
    std::vector<std::string> key_word = {
        "kill",
        "sudo",
        "unlink",
        "uninstall",
        "yum",
        "top",
    };
    for(auto &word:key_word)
    {
        auto pos = cmd.find(word);
        if(pos != std::string::npos)
            return false;
    }

    return true;
}
// 接收指令并执行
std::string ExcuteCommand(const std::string &cmd)
{
    std::cout << "get a request cmd: " << cmd << std::endl;
    if(!SafeCheck(cmd))
        return "Bad man";
    
    // popen函数会启动一个新的进程执行cmd命令，并将其输出与文件指针fp关联起来
    // 如果成功，fp将指向该命令输出的流，否则返回nullptr
    FILE* fp = popen(cmd.c_str(),"r");
    if(nullptr == fp)
    {
        perror("popen");
        return "error";
    }

    std::string result;
    char buffer[4096];
    while(true)
    {
        // 用buffer接收fp中的cmd输出结果，直到接收的数据为空，返回nullptr由in接收
        char *in = fgets(buffer,sizeof(buffer),fp);
        if(in == nullptr)
            break;
        
        result += buffer;
    }
    // 关闭由popen打开的文件流，并等待相关进程终止
    pclose(fp);

    return result;
}
int main(int argc,char* argv[])
{
    if(argc != 2)
    {
        Usage(argv[0]);
        exit(0);
    }
    uint16_t port = std::stoi(argv[1]);

    std::unique_ptr<UdpServer> svr(new UdpServer(port));

    svr->Init();

    // // 客户端向服务端发送指令并执行
    // svr->Run(ExcuteCommand);

    // 客户端向服务端发送消息
    svr->Run();

    return 0;
}